package com.itheima.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.OrderItem;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.itheima.mapper.EmpMapper;
import com.itheima.pojo.*;
import com.itheima.service.DeptService;
import com.itheima.service.EmpExprService;
import com.itheima.service.EmpService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.time.LocalDateTime;
import java.util.List;
import java.util.stream.Collectors;

@Service
public class EmpServiceImpl extends ServiceImpl<EmpMapper, Emp> implements EmpService {
    @Autowired
    private DeptService deptService;

    @Autowired
    private EmpExprService empExprService;

    /**
     * 条件分页查询
     *
     * @param param 查询参数
     * @return 分页结果
     */
    @Override
    public PageResult<Emp> getPageResult(EmpQueryParam param) {
        // 构建分页对象
        Page<Emp> page = new Page<>(param.getPage(), param.getPageSize());

        // 按照最后修改时间降序排序 asc: 升序 desc：降序
        page.addOrder(OrderItem.desc("update_time"));

        /*
        // 调用page方法完成分页
        LambdaQueryWrapper<Emp> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.like(param.getName() != null && !param.getName().isEmpty(), Emp::getName, param.getName())
                .eq(param.getGender() != null, Emp::getGender, param.getGender())
                .between(param.getBegin() != null && param.getEnd()!= null, Emp::getEntryDate, param.getBegin(), param.getEnd());

        page = page(page, queryWrapper);
        */
        // 使用lambdaQuery()来完成条件分页查询
        page = lambdaQuery().like(param.getName() != null && !param.getName().isEmpty(), Emp::getName, param.getName())
                .eq(param.getGender() != null, Emp::getGender, param.getGender())
                .between(param.getBegin() != null && param.getEnd() != null, Emp::getEntryDate, param.getBegin(), param.getEnd())
                .page(page);

        // 返回分页对象结果
        List<Emp> emps = page.getRecords();

        // 遍历员工集合，获取到所有员工的所有部门id的集合
        List<Long> deptIds = emps.stream().map(emp -> emp.getDeptId()).collect(Collectors.toList());

        if (!deptIds.isEmpty()) {
            // 调用service层方法，根据部门id集合批量查询部门信息（只查一次部门表）
            List<Dept> depts = deptService.listByIds(deptIds);

            // 在内存中将部门信息设置到员工对象中
            emps.forEach(emp -> {
                // 遍历部门集合，获取到当前员工的部门名称
                depts.forEach(dept -> {
                    if (emp.getDeptId() != null && dept.getId().equals(emp.getDeptId())) {
                        emp.setDeptName(dept.getName());
                    }
                });
            });
        }
        return new PageResult<>(page.getTotal(), emps);
    }

    /**
     * 保存员工基本信息和工作经历信息
     *
     * @param emp 员工数据
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void saveEmp(Emp emp) {
        // 1.保存员工基本信息
        emp.setCreateTime(LocalDateTime.now());
        save(emp);  // MP的保存方法会自定返回主键

        // 2.保存员工工作经历信息，需要为每一条工作经历添加emp_id
        List<EmpExpr> exprList = emp.getExprList();

        if (exprList == null || exprList.isEmpty()) {
            return;
        }
        exprList.forEach(expr -> expr.setEmpId(emp.getId()));

        // 批量保存工作经历信息
        empExprService.saveBatch(exprList);
    }

    /**
     * 根据id查询员工
     *
     * @param id 员工id
     * @return 员工信息
     */
    @Override
    public Emp getEmpById(Integer id) {
        // 1.查询员工基本信息
        Emp emp = getById(id);

        // 2.查询员工工作经历信息
        List<EmpExpr> empExprList = empExprService.list(new LambdaQueryWrapper<EmpExpr>().eq(EmpExpr::getEmpId, id));

        if (empExprList != null) {
            emp.setExprList(empExprList);
        }
        return emp;
    }

    /**
     * 修改员工信息
     *
     * @param emp 员工对象
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void updateEmp(Emp emp) {
        // 1.修改员工基本信息
        emp.setUpdateTime(LocalDateTime.now());
        updateById(emp);

        // 2.删除员工工作经历信息
        empExprService.remove(new LambdaQueryWrapper<EmpExpr>().eq(EmpExpr::getEmpId, emp.getId()));

        // 3.添加新的员工工作经历信息
        List<EmpExpr> exprList = emp.getExprList();

        if (exprList != null && !exprList.isEmpty()) {
            // 为每一条工作经历添加emp_id
            exprList.forEach(expr -> expr.setEmpId(emp.getId()));
            // 批量保存
            empExprService.saveBatch(exprList);
        }
    }

    /**
     * 批量删除员工
     *
     * @param ids 要删除的员工id集合
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void deleteByIds(List<Integer> ids) {
        // 1.删除员工基本信息
        removeBatchByIds(ids);

        // 2.删除员工工作经历信息
        empExprService.remove(new LambdaQueryWrapper<EmpExpr>().in(EmpExpr::getEmpId, ids));
    }
}
